package com.example.wu.recyclerviewtext;

import android.content.Intent;
import android.support.v7.app.AppCompatActivity;
import android.os.Bundle;
import android.support.v7.widget.LinearLayoutManager;
import android.support.v7.widget.RecyclerView;
import android.util.Log;
import android.view.View;
import android.widget.Button;

import java.util.ArrayList;

public class MainActivity extends AppCompatActivity {
    private RecyclerView rlvStu;
    private ArrayList<Student> studentArrayList = new ArrayList<>();
    private StuAdapter adapter;
    private Button btnAddStu;

    @Override
    protected void onCreate(Bundle savedInstanceState) {
        super.onCreate(savedInstanceState);
        setContentView(R.layout.activity_main);

        initStudentArrayList();
        initAddBtn();

        adapter = new StuAdapter(MainActivity.this, studentArrayList);

        rlvStu = findViewById(R.id.rlv_stu);
        rlvStu.setAdapter(adapter);

        LinearLayoutManager manager = new LinearLayoutManager(MainActivity.this);
        rlvStu.setLayoutManager(manager);
    }

    /**
     * 初始化添加按钮
     */
    private void initAddBtn() {
        btnAddStu = findViewById(R.id.btn_addStu);
        btnAddStu.setOnClickListener(new View.OnClickListener() {
            @Override
            public void onClick(View view) {
                Intent intent = new Intent();
                intent.setClass(MainActivity.this,AddActivity.class);
                startActivityForResult(intent,1001);
            }
        });
    }

    @Override
    protected void onActivityResult(int requestCode, int resultCode, Intent data) {
        Log.e("MainActivity",
                "requestCode = "+requestCode + "resultCode" + resultCode);

        if (data == null){
            return;
        }

        String num = data.getStringExtra("NUM");
        String name = data.getStringExtra("NAME");
        int age = data.getIntExtra("AGE",0);
        int imgId = data.getIntExtra("IMG",R.drawable.dog1);

        Student s = new Student(num, name, age, imgId);
        studentArrayList.add(s);
        adapter.notifyDataSetChanged();
    }

    /**
     * 初始化学生列表
     */
    private void initStudentArrayList() {
        Student s1 = new Student("17001","张一",18, R.drawable.dog1);
        Student s2 = new Student("17002","张二",19, R.drawable.dog2);
        Student s3 = new Student("17003","张三",20, R.drawable.dog3);
        Student s4 = new Student("17004","张四",21, R.drawable.dog4);
        Student s5 = new Student("17005","张五",22, R.drawable.dog5);

        studentArrayList.add(s1);
        studentArrayList.add(s2);
        studentArrayList.add(s3);
        studentArrayList.add(s4);
        studentArrayList.add(s5);
    }
}
